# -*- coding=utf-8 -*-
# 实现登陆和注册的逻辑
import re

from flask import current_app, session
from . import api
from flask import request, jsonify
from iHome.utils.response_code import RET
from iHome import redis_store, db
from iHome.models import User

#
# @api.route("/session", methods=["POST"])
# def login():
#     """登陆操作"""
#
#     # 获取参数和判断是否有值
#     data_dict = request.json
#     mobile = data_dict.get("mobile")
#     password = data_dict.get("password")
#
#     if not all([mobile, password]):
#         return jsonify(errno=RET.PARAMERR, errmsg="参数错误")
#
#     # 校验手机号码是否正确
#     if not re.match("^1[34578][0-9]{9}$", mobile):
#         return jsonify(errno= RET.PARAMERR, errmsg="请输入正确的手机号")
#     # 通过mobile查询到指定的用户
#     try:
#         user = User.query.filter(User.mobile==mobile).first()
#     except Exception as e:
#         current_app.logger.error(e)
#         return jsonify(errno=RET.DBERR, errmsg="查询数据失败")
#
#     if not user:
#         return jsonify(errno=RET.USERERR, errmsg="当前用户不存在")
#
#     # 校验密码
#     if not user.check_password(password):
#         return jsonify(errno=RET.PWDERR, errmsg="密码错误")
#     # 保存用户登录状态
#     session["user_id"]=user.id
#     session["name"]=user.name
#     session["mobile"]=user.mobile
#
#     # 返回结果
#     return jsonify(errno=RET.OK, errmsg="登陆成功")


@api.route("/session", methods=["DELETE"])
def logout():
    """
    1. 清除session中的对应登录之后保存的信息
    :return:
    """
    session.pop('user_id', None)
    session.pop('mobile', None)
    session.pop('name', None)

    return jsonify(errno=RET.OK, errmsg="OK")


@api.route("/session", methods=["POST"])
def login():
    """
    1. 获取参数和判断是否有值
    2. 从数据库查询出指定的用户
    3. 校验密码
    4. 保存用户登录状态
    5. 返回结果
    :return:
    """

    # 1. 获取参数和判断是否有值
    data_dict = request.json

    mobile = data_dict.get("mobile")
    password = data_dict.get("password")

    if not all([mobile, password]):
        return jsonify(errno=RET.PARAMERR, errmsg="参数不全")

    # 2. 从数据库查询出指定的用户
    try:
        user = User.query.filter_by(mobile=mobile).first()
    except Exception as e:
        current_app.logger.error(e)
        return jsonify(errno=RET.DBERR, errmsg="查询数据错误")

    if not user:
        return jsonify(errno=RET.USERERR, errmsg="用户不存在")

    # 3. 校验密码
    if not user.check_passowrd(password):
        return jsonify(errno=RET.PWDERR, errmsg="密码错误")

    # 4. 保存用户登录状态
    session["user_id"] = user.id
    session["name"] = user.name
    session["mobile"] = user.mobile

    # 5. 返回结果
    return jsonify(errno=RET.OK, errmsg="登录成功")


# @api.route("/users", methods=["POST"])
# def register():
#     """注册逻辑"""
#     # 获取参数并校验：手机号，短信验证码， 密码
#     data_dict = request.json
#
#     mobile = data_dict.get("mobile")
#     phonecode = data_dict.get("phonecode")
#     password = data_dict.get("password")
#
#     if not all([mobile, phonecode, password]):
#         return jsonify(errno=RET.PARAMERR, errmsg="参数错误")
#
#     # 取到真实的短信验证码
#     try:
#         real_phonecode = redis_store.get("Mobile:" + mobile)
#     except Exception as e:
#         current_app.logger.error(e)
#         return jsonify(errno=RET.DBERR, errmsg="查询验证码失败")
#
#     if not real_phonecode:
#         return jsonify(errno=RET.NODATA, errmsg="短信验证码过期")
#     # 进行验证码的对比
#     if phonecode != real_phonecode:
#         return jsonify(errno=RET.DATAERR, errmsg="短信验证码输入错误")
#
#     # 判断当前手机号是否已经被注册
#     if User.query.filter(User.mobile == mobile).first():
#         return jsonify(errno=RET.DATAEXIST, errmsg="当前手机号已经被注册")
#
#     # 初始化USER模型， 保存相关数据
#     user = User()
#     user.name = mobile
#     user.mobile = mobile
#     # 保存密码
#     user.password = password
#
#     # 将user模型保存到数据库中
#     try:
#         db.session.add(user)
#         db.session.commit()
#     except Exception as e:
#         db.session.rollback()
#         current_app.logger.error(e)
#         return jsonify(errno=RET.DATAERR, errmsg="保存用户数据失败")
#
#     # B保存当前用户信息到session中
#     session["user_id"] = user.id
#     session["name"] = user.name
#     session["mobile"] = user.mobile
#     # 给出响应：{"errno":"0", "errmsg":"注册成功"}
#     return jsonify(errno=RET.OK, errmsg="保存成功")

@api.route("/user", methods=["POST"])
def register():
    """
    1. 获取参数和判断是否有值
    2. 从redis中获取指定手机号对应的短信验证码的
    3. 校验验证码
    4. 初始化 user 模型，并设置数据并添加到数据库
    5. 保存当前用户的状态
    6. 返回注册的结果
    :return:
    """

    # 1. 获取参数和判断是否有值
    data_dict = request.json
    mobile = data_dict.get("mobile")
    phonecode = data_dict.get("phonecode")
    password = data_dict.get("password")

    if not all([mobile, phonecode, password]):
        return jsonify(errno=RET.PARAMERR, errmsg="参数不全")

    # 2. 从redis中获取指定手机号对应的短信验证码的
    try:
        sms_code = redis_store.get("SMS_" + mobile)
    except Exception as e:
        current_app.logger.error(e)
        return jsonify(errno=RET.DBERR, errmsg="获取本地验证码失败")

    if not sms_code:
        return jsonify(errno=RET.NODATA, errmsg="短信验证码过期")

    # 3. 校验验证码
    if phonecode != sms_code:
        return jsonify(errno=RET.DATAERR, errmsg="短信验证码错误")

    # 4. 初始化 user 模型，并设置数据并添加到数据库
    user = User()
    user.name = mobile
    user.mobile = mobile
    # 对密码进行处理
    user.password = password

    try:
        db.session.add(user)
        db.session.commit()
    except Exception as e:
        db.session.rollback()
        current_app.logger.error(e)
        return jsonify(errno=RET.DATAERR, errmsg="数据保存错误")

    session["user_id"] = user.id
    session["name"] = user.name
    session["mobile"] = user.mobile

    return jsonify(errno=RET.OK, errmsg="注册成功")
